package Compation;

import java.util.*;

/**
 * 1331. 数组序号转换
 * 给你一个整数数组 arr ，请你将数组中的每个元素替换为它们排序后的序号。
 *
 * 序号代表了一个元素有多大。序号编号的规则如下：
 *
 * 序号从 1 开始编号。
 * 一个元素越大，那么序号越大。如果两个元素相等，那么它们的序号相同。
 * 每个数字的序号都应该尽可能地小。
 *
 *
 * 示例 1：
 *
 * 输入：arr = [40,10,20,30]
 * 输出：[4,1,2,3]
 * 解释：40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。
 * 示例 2：
 *
 * 输入：arr = [100,100,100]
 * 输出：[1,1,1]
 * 解释：所有元素有相同的序号。
 * 示例 3：
 *
 * 输入：arr = [37,12,28,9,100,56,80,5,12]
 * 输出：[5,3,4,2,8,6,7,1,3]
 *
 *
 * 提示：
 *
 * 0 <= arr.length <= 105
 * -109 <= arr[i] <= 109
 */
public class arrayRankTransform {
    /**
     * 方法一：利用Hashset/TreeSet，会超时
     * @param arr
     * @return
     */
    public int[] arrayRankTransform(int[] arr) {
        int[] result = new int[arr.length];
        if (arr == null || arr.length == 0) {
            return result;
        }
        Set<Integer> numSet = new HashSet<>();
        for (int i : arr) {
            numSet.add(i);
        }
        List<Integer> numList = new ArrayList<>(numSet);
        Collections.sort(numList);
        for (int i = 0; i < arr.length; i++) {
            result[i] = numList.indexOf(arr[i]) + 1;
        }
        return result;
    }

    /**
     * 方法二：直接对输入数组排序
     * @param arr
     * @return
     */
    // TODO: 存在问题
    public int[] arrayRankTransformI(int[] arr) {
        int[] result = new int[arr.length];
        if (arr == null || arr.length == 0) {
            return result;
        }
        Arrays.sort(arr);
        Map<Integer, Integer> map = new HashMap<>();
        int index = 1;
        for (int i = 0; i < arr.length; i++) {
            if (i == 0) {
                map.put(arr[i], index);
                index++;
            }
            if (i > 0 && arr[i] != arr[i - 1]) {
                map.put(arr[i], index);
                index++;
            }
        }
        for (int i = 0; i < arr.length; i++) {
            result[i] = map.get(arr[i]);
        }
        return result;
    }
}
